package com.zhou.service.impl;

import com.alibaba.dubbo.config.annotation.Service;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.zhou.constant.MessageConstant;
import com.zhou.dao.PermissionMapper;
import com.zhou.dao.RoleMapper;
import com.zhou.entity.PageResult;
import com.zhou.entity.QueryPageBean;
import com.zhou.entity.Result;
import com.zhou.pojo.Permission;
import com.zhou.pojo.Role;
import com.zhou.service.RoleService;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service(interfaceClass = RoleService.class)
public class RoleServiceImpl implements RoleService {
    @Autowired
    private RoleMapper roleMapper;
    @Autowired
    private PermissionMapper permissionMapper;

    /**
     * 查询所有角色
     *
     * @return
     */
    @Override
    public List<Role> findAll() {
        return roleMapper.findAll();
    }

    /**
     * 新增角色，同时需要让用户关联权限，关联菜单
     *
     * @param role
     * @param permissionIds
     * @param menuIds
     * @return
     */
    @Override
    public Result add(Role role, Integer[] permissionIds, Integer[] menuIds) {
        //先判断角色是否存在
        long count = roleMapper.findExist(role);
        if (count > 0) {
            return new Result(false, "该角色已经存在,无法创建");
        }
        roleMapper.insert(role);
        //关联角色和权限的关系表
        setRoleAndPermission(role.getId(), permissionIds);
        //关联菜单角色的中间表
        for (Integer menuId : menuIds) {
            roleMapper.addRoleMenu(role.getId(), menuId);
        }
        return new Result(true, MessageConstant.ADD_ROLE_SUCCESS);
    }

    /**
     * 建立角色和权限多对多关系
     *
     * @param roleId
     * @param permissionIds
     */
    private void setRoleAndPermission(Integer roleId, Integer[] permissionIds) {
        if (permissionIds != null && permissionIds.length > 0) {
            Map<String, Integer> map = null;
            for (Integer permissionId : permissionIds) {
                map = new HashMap<>();
                map.put("roleId", roleId);
                map.put("permissionId", permissionId);
                roleMapper.setRoleAndPermission(map);
            }
        }
    }

    /**
     * 根据菜单id查询所有的权限信息
     *
     * @param id
     * @return
     */
    @Override
    public Map findByIdWithAllPermission(Integer id) {
        //根据角色id查询 角色的基本信息
        Role formData = roleMapper.selectByPrimaryKey(id);
        //查询所有的权限信息
        List<Permission> tableData = permissionMapper.findAll();
        //根据菜单id查询权限id
        List<Integer> permissionIds = permissionMapper.findPermissionById(id);
        Map<String, Object> map = new HashMap<>();
        map.put("formData", formData);
        map.put("tableData", tableData);
        map.put("permissionIds", permissionIds);
        return map;
    }

    @Override
    public void edit(Role role, Integer[] permissionIds, Integer[] menuIds) {
        //修改角色表基本信息，操作检查组t_role表
        roleMapper.updateByPrimaryKeySelective(role);
        //清理当前角色关联的权限，操作中间关系表 t_role_permission 表
        roleMapper.deleteAssociation(role.getId());
        //重新建立当前角色和权限的关联关系，操作 t_role_permission 表
        Integer roleId = role.getId();
        setRoleAndPermission(roleId, permissionIds);
        //删除旧的角色菜单关系
        roleMapper.deleteRoleMenuByRoleId(roleId);
        for (Integer menuId : menuIds) {
            //添加新的角色菜单关系
            roleMapper.addRoleMenu(roleId, menuId);
        }
    }

    @Override
    public Result deleteById(Integer id) {
        //判断该角色是否是系统管理员
        Role role = roleMapper.selectByPrimaryKey(id);
        if (role.getKeyword().equals("ROLE_ADMIN")) {
            return new Result(false, "该角色为系统管理员，不可删除！");
        }
        //判断角色是否被用户所有关联
        Long count = roleMapper.findUserByRole(id);
        if (count > 0) {
            return new Result(false, "该角色已被用户关联，无法删除！");
        }
        //删除角色和权限的关联表(t_role_permission)的数据,根据用户id进行删除
        roleMapper.deleteAssociation(id);
        //在删除角色菜单关系
        roleMapper.deleteRoleMenuByRoleId(id);
        //删除角色
        roleMapper.deleteByPrimaryKey(id);
        return new Result(true, MessageConstant.DELETE_ROLE_SUCCESS);
    }

    @Override
    public PageResult findByPage(QueryPageBean queryPageBean) {
        PageHelper.startPage(queryPageBean.getCurrentPage(), queryPageBean.getPageSize());
        Page<Role> page = roleMapper.findByPage(queryPageBean.getQueryString());
        return new PageResult(page.getTotal(), page.getResult());
    }
}
